# -*- coding: utf-8 -*-
"""
@Time ： 2020/9/25 16:22
@Auth ： Mr. JJ_Cloud
@Company ：testingedu.com.cn
@Function ：app 反复启动
"""
import os
import time


class AppRebooter():
    # 初始化方法的作用在于：当每个实例对象创建时，该方法内的代码无须调用就会自动运行
    def __init__(self, package_name, activity_name):
        self.package_name = package_name
        self.activity_name = activity_name

    def reboot_app(self):
        os.system("adb shell am start %s/%s" % (self.package_name, self.activity_name))
        os.system("adb shell screencap -p /sdcard/a.png")
        os.system("adb pull /sdcard/a.png D:\\Test\\once_%s.png" % i)  # App启动中的截图
        time.sleep(5)
        os.system("adb shell screencap -p /sdcard/a.png")
        os.system("adb pull /sdcard/a.png D:\\Test\\twice_%s.png" % i)  # App启动完成的截图
        os.system("adb shell am force-stop %s" % self.package_name)


if __name__ == '__main__':
    a_obj = AppRebooter("me.niuke.app.gupiaozhitou", "tv.aniu.dzlc.anzt.SplashActivity")
    for i in range(10):
        print("正在进行第%s轮重启测试" % i)
        a_obj.reboot_app()
    os.system("pause") # 请按任意键继续...